#pragma once
#include <iostream>
#include <string>
#include <queue>
#include <vector>
#include <memory>
#include <functional>
#include "Log.hpp"
#include "Thread.hpp"
#include "Mutex.hpp"
#include "Cond.hpp"

namespace ThreadPoolModule
{
    using namespace ThreadModule;
    using namespace LogMudule;
    using namespace LockModule;
    using namespace CondModule;

    using thread_t = std::shared_ptr<Thread>;

    void DefaultTest()
    {
        while (true)
        {
            LOG(LogLevel::INFO) << "测试方法";
            sleep(1);
        }
    }

    const static int defaultnum = 5;

    template <typename T>
    class ThreadPool
    {
    private:
        bool IsEmpty() { return _taskq.empty(); }

        void HandlerTask(std::string name)
        {
            LOG(LogLevel::INFO) << "线程:" << name << "进入HandlerTask的逻辑";
            while (true)
            {

                // 1.拿任务
                T t;
                {
                    Lockguard lockguard(_lock);
                    while (IsEmpty() && _isrunning)
                    {
                        _wait_num++;
                        _cond_Wait.wait(_lock);
                        _wait_num--;
                    }
                    if (IsEmpty() && !_isrunning)
                        break;
                    t = _taskq.front();
                    _taskq.pop();
                }
                // 2.处理任务
                t(); // 规定，未来所有的任务处理，全部都是必须提供()方法！
            }
        }
        ThreadPool(const ThreadPool<T> &) = delete;
        ThreadPool<T> &operator=(const ThreadPool<T> &) = delete;

        ThreadPool(int num = defaultnum) : _num(num), _wait_num(0)
        {
            for (int i = 0; i < _num; i++)
            {

                //_threads.push_back(std::make_shared<Thread>(std::bind(&ThreadPool::HandlerTask,this,std::placeholders::_1)));
                _threads.push_back(std::make_shared<Thread>([this](std::string name)
                                                            { this->HandlerTask(name); })); // lambda表达式的类型是根据返回值和参数决定的
                       //注意两点吧：因为在Thread中的func_t是std::function<void(std::string name)>类型的，所以我们回调的函数也要传name，即 
                       //HandlerTask()要传name，第二点是lambda的圆括号是可调用对象HandlerTask需要传的参数，this是捕捉的当前成员变量                                                              
                       // 内部封装了operator()，传给thread构造线程对象
                       // 可以直接func()来进行回调函数

                //构造这些线程，每个线程绑定的函数是HandlerTask(name)，在HandlerTask(name)中拿的t就是转发模块中的Route方法
                LOG(LogLevel::INFO) << "构建线程" << _threads.back()->Name() << "对象 ...成功";
            }
        }

    public:
        static ThreadPool<T> *getInstance()
        {
            if (instance == NULL)
            {
                Lockguard lockguard(mutex);
                if (instance == NULL)
                {
                    LOG(LogLevel::INFO) << "单例首次被执行，需要加载对象...";
                    instance = new ThreadPool<T>();
                    instance->Start();
                }
            }

            return instance;
        }

        void Equeue(T &in)
        {
            Lockguard lockguard(_lock);
            _taskq.push(in); // 将右值转换成左值，避免了拷贝，直接交换资源
            if (_wait_num)
                _cond_Wait.Notify();
        }

        void Start()
        {
            if (_isrunning)
                return;
            _isrunning = true; // bug fix??
            for (auto &thread_ptr : _threads)
            {
                LOG(LogLevel::INFO) << "启动线程" << thread_ptr->Name() << " ... 成功";
                thread_ptr->Start();
            }
        }

        void Wait()
        {
            for (auto &thread_ptr : _threads)
            {
                thread_ptr->Join();
                LOG(LogLevel::INFO) << "停止线程" << thread_ptr->Name() << "对象 ...成功";
            }
        }

        void Stop()
        {
            Lockguard lockguard(_lock);
            if (_isrunning)
            {
                _isrunning = false;

                if (_wait_num)
                    _cond_Wait.NotifyAll(); // 唤醒所有等待的线程
            }
        }

        ~ThreadPool()
        {
        }

    private:
        int _num;
        std::queue<T> _taskq; // 任务队列里放的任务由模板决定
        std::vector<thread_t> _threads;
        bool _isrunning = false;

        int _wait_num;

        Mutex _lock;
        Cond _cond_Wait;

        static ThreadPool<T> *instance; // 声明
        static Mutex mutex;             // 只用来保护单例
    };
    // 这里得再次加上template，因为是在类外进行初始化的需要遵循语法规则，类的类型是ThreadPool<T>
    template <typename T>
    ThreadPool<T> *ThreadPool<T>::instance = NULL; // 定义
    template <typename T>
    LockModule::Mutex ThreadPool<T>::mutex; // 只用来保护单例

}